//
// Created by alexa on 5/5/2024.
//

#ifndef ARTILLERYSYSTEM_BATTALION_H
#define ARTILLERYSYSTEM_BATTALION_H


#include <vector>
#include "../_identifiable/Identifiable.h"

using namespace std;

class Battalion : public Identifiable {
private:
    vector<string> unitsIds;
    string name;
public:

    Battalion(const string &name);

    vector<string> getAllUnitsIds();

    void removeUnitId(string unitId);

    void addUnitId(string unitId);

    const string &getName() const;

    Battalion& operator+=(string unitId);

    Battalion& operator-=(string unitId);
};


#endif //ARTILLERYSYSTEM_BATTALION_H
